public class code4 {
    public int maxProfit(int[] prices) {
        int INF = 0x3f3f3f3f;
        int n = prices.length;

        int[][] f = new int[n][3];//i和j分别表示：持有股票+交易次数j次
        int[][] g = new int[n][3];

        for(int i=1;i<3;i++) {
            f[0][i] = g[0][i] = -INF;
        }
        f[0][0] = -prices[0];

        for(int i=1;i<n;i++) {
            for(int j=0;j<3;j++) {
                f[i][j]=Math.max(f[i-1][j],g[i-1][j]-prices[i]);
                //g[i][j]=Math.max(g[i-1][j],f[i-1][j-1]+prices[i]);
                g[i][j]=g[i-1][j];
                if(j-1>=0) g[i][j]=Math.max(g[i][j],f[i-1][j-1]+prices[i]);//防止j为-1
            }
        }

        //返回值：g表最后一行的最大值
        int ret = g[n-1][0];
        for(int i=1;i< 3;i++) ret = Math.max(ret,g[n-1][i]);
        return ret;

    }
}
